home *** CD-ROM | disk | FTP | other *** search
/ Programming an RTS Game with Direct3D / Programming an RTS Game with Direct3D.iso / Examples / Chapter 14 / Example 14.1 / mouse.cpp < prev    next >
Encoding:
C/C++ Source or Header  |  2006-07-16  |  7.2 KB  |  284 lines
  1. #include "mouse.h"
  2.  
  3. //////////////////////////////////////////////////////////////////////////
  4. //                        RAY                                                //
  5. //////////////////////////////////////////////////////////////////////////
  6.  
  7. RAY::RAY()
  8. {
  9.     org = dir = D3DXVECTOR3(0.0f, 0.0f, 0.0f);
  10. }
  11.  
  12. RAY::RAY(D3DXVECTOR3 o, D3DXVECTOR3 d)
  13. {
  14.     org = o;
  15.     dir = d;
  16. }
  17.  
  18. float RAY::Intersect(MESHINSTANCE iMesh)
  19. {
  20.     if(iMesh.m_pMesh == NULL)return -1.0f;
  21.     return Intersect(iMesh.m_pMesh->m_pMesh);
  22. }
  23.  
  24. float RAY::Intersect(BBOX bBox)
  25. {
  26.     if(D3DXBoxBoundProbe(&bBox.min, &bBox.max, &org, &dir))
  27.         return D3DXVec3Length(&(((bBox.min + bBox.max) / 2.0f) - org));
  28.     else return -1.0f;
  29. }
  30.  
  31. float RAY::Intersect(BSPHERE bSphere)
  32. {
  33.     if(D3DXSphereBoundProbe(&bSphere.center, bSphere.radius, &org, &dir))
  34.         return D3DXVec3Length(&(bSphere.center - org));
  35.     else return -1.0f;
  36. }
  37.  
  38. float RAY::Intersect(ID3DXMesh* mesh)
  39. {
  40.     if(mesh == NULL)return -1.0f;
  41.  
  42.     // Collect only the closest intersection
  43.     BOOL hit;
  44.     DWORD dwFace;
  45.     float hitU, hitV, dist;
  46.     D3DXIntersect(mesh, &org, &dir, &hit, &dwFace, &hitU, &hitV, &dist, NULL, NULL);
  47.  
  48.     if(hit) return dist;
  49.     else return -1.0f;
  50. }
  51.  
  52. //////////////////////////////////////////////////////////////////////////
  53. //                        MOUSE                                            //
  54. //////////////////////////////////////////////////////////////////////////
  55.  
  56. MOUSE::MOUSE()
  57. {
  58.     m_iType = 0;
  59.     m_pMouseTexture = NULL;
  60.     m_pMouseDevice = NULL;
  61.     m_fSpeed = 1.5f;
  62.     m_dwBlock = GetTickCount();
  63. }
  64.  
  65. MOUSE::~MOUSE()
  66. {
  67.     if(m_pMouseDevice != NULL)
  68.         m_pMouseDevice->Release();
  69.  
  70.     if(m_pMouseTexture != NULL)
  71.         m_pMouseTexture->Release();
  72. }
  73.  
  74. void MOUSE::InitMouse(IDirect3DDevice9* Dev, HWND wnd)
  75. {
  76.     try
  77.     {
  78.         m_pDevice = Dev;
  79.  
  80.         //Load texture and init sprite
  81.         D3DXCreateTextureFromFile(m_pDevice, "cursor/cursor.dds", &m_pMouseTexture);
  82.         D3DXCreateSprite(m_pDevice, &m_pSprite);
  83.  
  84.         //Get directinput object
  85.         LPDIRECTINPUT8 directInput = NULL;
  86.  
  87.         DirectInput8Create(GetModuleHandle(NULL),    // Retrieves the application handle
  88.                            0x0800,                    // v.8.0    
  89.                            IID_IDirectInput8,        // Interface ID
  90.                            (VOID**)&directInput,    // Our DirectInput Object
  91.                            NULL);
  92.  
  93.         //Acquire Default System mouse
  94.         directInput->CreateDevice(GUID_SysMouse, &m_pMouseDevice, NULL);        
  95.         m_pMouseDevice->SetDataFormat(&c_dfDIMouse);
  96.         m_pMouseDevice->SetCooperativeLevel(wnd, DISCL_EXCLUSIVE | DISCL_FOREGROUND);
  97.         m_pMouseDevice->Acquire();            
  98.  
  99.         //Get viewport size and init mouse location
  100.         D3DVIEWPORT9 v;
  101.         m_pDevice->GetViewport(&v);
  102.         m_viewport.left = v.X;
  103.         m_viewport.top = v.Y;
  104.         m_viewport.right = v.X + v.Width;
  105.         m_viewport.bottom = v.Y + v.Height;
  106.  
  107.         x = v.X + v.Width / 2.0f;
  108.         y = v.Y + v.Height / 2.0f;
  109.  
  110.         //Release Direct Input Object
  111.         directInput->Release();
  112.  
  113.         //Create Mouse Sphere
  114.         D3DXCreateSphere(m_pDevice, 0.2f, 5, 5, &m_pSphereMesh, NULL);
  115.  
  116.         //Create Ball Material
  117.         m_ballMtrl.Diffuse = D3DXCOLOR(1.0f, 1.0f, 0.0f, 1.0f);
  118.         m_ballMtrl.Specular = m_ballMtrl.Ambient = m_ballMtrl.Emissive = D3DXCOLOR(0.0f, 0.0f, 0.0f, 1.0f);
  119.     }
  120.     catch(...)
  121.     {
  122.         debug.Print("Error in MOUSE::InitMouse()");
  123.     }    
  124. }
  125.  
  126. bool MOUSE::ClickLeft()
  127. { 
  128.     if(GetTickCount() < m_dwBlock)return false;
  129.     return m_mouseState.rgbButtons[0];
  130. }
  131.  
  132. bool MOUSE::ClickRight()
  133. {
  134.     if(GetTickCount() < m_dwBlock)return false;
  135.     return m_mouseState.rgbButtons[1];
  136. }
  137.  
  138. bool MOUSE::WheelUp()
  139. {
  140.     if(GetTickCount() < m_dwBlock)return false;
  141.     return m_mouseState.lZ > 0.0f;
  142. }
  143.  
  144. bool MOUSE::WheelDown()
  145. {
  146.     if(GetTickCount() < m_dwBlock)return false;
  147.     return m_mouseState.lZ < 0.0f;
  148. }
  149.  
  150. bool MOUSE::Over(RECT dest)
  151. {
  152.     if(x < dest.left || x > dest.right)return false;
  153.     if(y < dest.top || y > dest.bottom)return false;
  154.     return true;
  155. }
  156.  
  157. bool MOUSE::PressInRect(RECT dest)
  158. {
  159.     return ClickLeft() && Over(dest);
  160. }
  161.  
  162. void MOUSE::Update(TERRAIN *terrain)
  163. {
  164.     //Retrieve mouse state
  165.     ZeroMemory(&m_mouseState, sizeof(DIMOUSESTATE));
  166.     m_pMouseDevice->GetDeviceState(sizeof(DIMOUSESTATE), &m_mouseState);
  167.  
  168.     //Update pointer
  169.     x += m_mouseState.lX * m_fSpeed;
  170.     y += m_mouseState.lY * m_fSpeed;
  171.  
  172.     //Keep mouse pointer within window
  173.     if(x < m_viewport.left)    x = m_viewport.left;
  174.     if(y < m_viewport.top)    y = m_viewport.top;
  175.     if(x > m_viewport.right)    x = m_viewport.right;
  176.     if(y > m_viewport.bottom)    y = m_viewport.bottom;
  177.  
  178.     if(terrain != NULL)
  179.         CalculateMappos(*terrain);
  180. }
  181.  
  182. void MOUSE::Paint()
  183. {    
  184.     //Draw ball
  185.     D3DXMATRIX world;
  186.     D3DXMatrixTranslation(&world, m_ballPos.x, m_ballPos.y, m_ballPos.z);
  187.     m_pDevice->SetTransform(D3DTS_WORLD, &world);
  188.     m_pDevice->SetMaterial(&m_ballMtrl);
  189.     m_pDevice->SetTexture(0, NULL);
  190.     m_pSphereMesh->DrawSubset(0);
  191.  
  192.     if(m_pMouseTexture != NULL)
  193.     {
  194.         RECT src[] = {{0,0,20,20}, {0,20,20,40}, {20,20,40,40}, {0,40,20,60}, {20,40,40,60}};
  195.  
  196.         m_pSprite->Begin(D3DXSPRITE_ALPHABLEND);
  197.         m_pSprite->Draw(m_pMouseTexture, &src[m_iType], NULL, &D3DXVECTOR3(x, y, 0.0f), 0xffffffff);
  198.         m_pSprite->End();
  199.     }    
  200. }
  201.  
  202. RAY MOUSE::GetRay()
  203. {
  204.     try
  205.     {
  206.         D3DXMATRIX projectionMatrix, viewMatrix, worldViewInverse, worldMatrix;
  207.         
  208.         m_pDevice->GetTransform(D3DTS_PROJECTION, &projectionMatrix);
  209.         m_pDevice->GetTransform(D3DTS_VIEW, &viewMatrix);
  210.         m_pDevice->GetTransform(D3DTS_WORLD, &worldMatrix);
  211.  
  212.         float width = m_viewport.right - m_viewport.left;
  213.         float height = m_viewport.bottom - m_viewport.top;
  214.  
  215.         float angle_x = (((2.0f * x) / width) - 1.0f) / projectionMatrix(0,0);
  216.         float angle_y = (((-2.0f * y) / height) + 1.0f) / projectionMatrix(1,1); 
  217.         
  218.         RAY ray;
  219.         ray.org = D3DXVECTOR3(0.0f, 0.0f, 0.0f);
  220.         ray.dir = D3DXVECTOR3(angle_x, angle_y, 1.0f);
  221.  
  222.         D3DXMATRIX m = worldMatrix * viewMatrix;
  223.         D3DXMatrixInverse(&worldViewInverse, 0, &m);
  224.         D3DXVec3TransformCoord(&ray.org, &ray.org, &worldViewInverse);
  225.         D3DXVec3TransformNormal(&ray.dir, &ray.dir, &worldViewInverse);
  226.         D3DXVec3Normalize(&ray.dir, &ray.dir);
  227.  
  228.         return ray;
  229.     }
  230.     catch(...)
  231.     {
  232.         debug.Print("Error in MOUSE::GetRay()");
  233.     }
  234. }
  235.  
  236. void MOUSE::CalculateMappos(TERRAIN &terrain)
  237. {
  238.     //Get Mouse Ray
  239.     D3DXMATRIX world;
  240.     D3DXMatrixIdentity(&world);
  241.     m_pDevice->SetTransform(D3DTS_WORLD, &world);
  242.     RAY mRay = GetRay();
  243.  
  244.     float minDistance = 10000.0f;
  245.     for(int i=0;i<terrain.m_patches.size();i++)
  246.     {
  247.         if(mRay.Intersect(terrain.m_patches[i]->m_BBox) > 0.0f)
  248.         {
  249.             // Collect only the closest intersection
  250.             BOOL hit;
  251.             DWORD dwFace;
  252.             float hitU, hitV, dist;
  253.             D3DXIntersect(terrain.m_patches[i]->m_pMesh, &mRay.org, &mRay.dir, &hit, &dwFace, &hitU, &hitV, &dist, NULL, NULL);
  254.  
  255.             if(hit && dist < minDistance)
  256.             {
  257.                 minDistance = dist;
  258.                 int tiles = dwFace / 2;        //Two faces to each map tile
  259.                 int tilesPerRow = terrain.m_patches[i]->m_mapRect.right - terrain.m_patches[i]->m_mapRect.left;
  260.                 int y = tiles / tilesPerRow, x = tiles - y * tilesPerRow;
  261.  
  262.                 if(dwFace % 2 == 0)        //Hit upper left face
  263.                 {
  264.                     if(hitU > 0.5f)x++;
  265.                     else if(hitV > 0.5f)y++;
  266.                 }
  267.                 else                    //Hit lower right face
  268.                 {
  269.                     if(hitU + hitV < 0.5f)y++;
  270.                     else if(hitU > 0.5f)x++;
  271.                     else {x++;y++;}
  272.                 }
  273.  
  274.                 m_mappos.Set(terrain.m_patches[i]->m_mapRect.left + x, terrain.m_patches[i]->m_mapRect.top + y);
  275.                 m_ballPos = terrain.GetWorldPos(m_mappos);
  276.             }            
  277.         }
  278.     }    
  279. }
  280.  
  281. void MOUSE::DisableInput(int millisec)
  282. {
  283.     m_dwBlock = GetTickCount() + millisec;
  284. }